# -- coding: utf-8 --

if __name__ == '__main__':

    # 判断img目录是否存在
    import os
    if not os.path.exists('img'):
        os.mkdir('img')
    from selenium import webdriver
    driver = webdriver.Firefox()
    driver.get_screenshot_as_file('img/a.png')